import numpy as np
import matplotlib.pyplot as plt

# 设置中文显示
plt.rcParams['font.sans-serif'] = 'SimHei'
plt.rcParams['axes.unicode_minus'] = False
inputfile = './data/国民经济核算季度数据.npz'
data = np.load(inputfile, allow_pickle=True)
# 提取其中的column数组，视为数据的标签
name = data['columns']
# 提取其中的values数组，数据的存在位置
values = data['values']

# 将画布设定为正方形，则绘制的饼图是正圆
plt.figure(figsize=(8, 7))
plt.rc('font', size=12)
# 绘制散点图
plt.scatter(values[:, 0], values[:, 3], marker='o', c='red') # 第一产业
plt.scatter(values[:, 0], values[:, 4], marker='D', c='blue') # 第二产业
plt.scatter(values[:, 0], values[:, 5], marker='v', c='yellow') # 第三产业
# 添加横轴标签
plt.xlabel('年份')
# 添加纵轴标签
plt.ylabel('生产总值（亿元）')
plt.xticks(range(0,81,4), values[range(0,81,4), 1], rotation=45)
plt.title('2000-2020年各产业第一季度生产总值散点图（徐伟杰）')
plt.legend(['第一产业', '第二产业', '第三产业']) # 添加图例
plt.show()


